package com.aries.common.core.mybatisplus.core;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.IService;

import java.io.Serializable;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.function.Function;

/**
 * 自定义 Service 接口, 实现 数据库实体与 vo 对象转换返回
 *
 * @param <T> 数据实体类
 * @param <V> vo类
 * @author Lion Li
 */
public interface IServicePlus<T, V> extends IService<T> {

    /**
     * @param id 主键id
     * @return V对象
     */
    V getVoById(Serializable id);

    /**
     * @param convertor 自定义转换器
     */
    default V getVoById(Serializable id, Function<T, V> convertor) {
        return convertor.apply(getById(id));
    }

    /**
     * @param idList id列表
     * @return V对象
     */
    List<V> listVoByIds(Collection<? extends Serializable> idList);

    /**
     * @param convertor 自定义转换器
     */
    default List<V> listVoByIds(Collection<? extends Serializable> idList,
                                Function<Collection<T>, List<V>> convertor) {
        List<T> list = getBaseMapper().selectBatchIds(idList);
        if (list == null) {
            return null;
        }
        return convertor.apply(list);
    }

    /**
     * @param columnMap 表字段 map 对象
     * @return V对象
     */
    List<V> listVoByMap(Map<String, Object> columnMap);

    /**
     * @param convertor 自定义转换器
     */
    default List<V> listVoByMap(Map<String, Object> columnMap,
                                Function<Collection<T>, List<V>> convertor) {
        List<T> list = getBaseMapper().selectByMap(columnMap);
        if (list == null) {
            return null;
        }
        return convertor.apply(list);
    }

    /**
     * @param queryWrapper 查询条件
     * @return V对象
     */
    V getVoOne(Wrapper<T> queryWrapper);

    /**
     * @param convertor 自定义转换器
     */
    default V getVoOne(Wrapper<T> queryWrapper, Function<T, V> convertor) {
        return convertor.apply(getOne(queryWrapper, true));
    }

    /**
     * @param queryWrapper 查询条件
     * @return V对象
     */
    List<V> listVo(Wrapper<T> queryWrapper);

    /**
     * @param convertor 自定义转换器
     */
    default List<V> listVo(Wrapper<T> queryWrapper, Function<Collection<T>, List<V>> convertor) {
        List<T> list = getBaseMapper().selectList(queryWrapper);
        if (list == null) {
            return null;
        }
        return convertor.apply(list);
    }

    default List<V> listVo() {
        return listVo(Wrappers.emptyWrapper());
    }

    /**
     * @param convertor 自定义转换器
     */
    default List<V> listVo(Function<Collection<T>, List<V>> convertor) {
        return listVo(Wrappers.emptyWrapper(), convertor);
    }

    boolean saveAll(Collection<T> entityList);

    boolean saveOrUpdateAll(Collection<T> entityList);
}

